package com.cxmx.escort.aspect.api;

public final class ProtocolApi {
    public static final class RequestKey {
        public static final String INPUT = "input";
    }

    public static final class ResponseKey {
        public static final String PREDICTED_LABEL = "predicted_label";
        public static final String CONFIDENCE_SCORE = "confidence_score";
        public static final String GRAPHS = "graphs";
        public static final class GraphKey {
            public static final String NAME = "name";
            public static final String URL = "url";
        }
    }

    public static class GraphInfo {
        public final String graphName;
        public final String imageUrl;

        public GraphInfo(String graphName, String imageUrl) {
            this.graphName = graphName;
            this.imageUrl = imageUrl;
        }

        public String toString() {
            return "  { name: " +
                    graphName +
                    "\n" +
                    "    url: " +
                    imageUrl +
                    "\n" +
                    "  }";
        }
    }
}
